import json
from tencentcloud.common import credential
from tencentcloud.common.profile.client_profile import ClientProfile
from tencentcloud.common.profile.http_profile import HttpProfile
from tencentcloud.common.exception.tencent_cloud_sdk_exception import TencentCloudSDKException
from tencentcloud.cdn.v20180606 import cdn_client, models


class Tencent:
    client = None

    def __init__(self, args):
        SecretId = args['AccessKeyId']
        SecretKey = args['AccessKeySecret']

        # 实例化一个认证对象，入参需要传入腾讯云账户 SecretId 和 SecretKey，此处还需注意密钥对的保密
        # 代码泄露可能会导致 SecretId 和 SecretKey 泄露，并威胁账号下所有资源的安全性。以下代码示例仅供参考，建议采用更安全的方式来使用密钥，请参见：https://cloud.tencent.com/document/product/1278/85305
        # 密钥可前往官网控制台 https://console.cloud.tencent.com/cam/capi 进行获取
        cred = credential.Credential(SecretId, SecretKey)
        # 实例化一个http选项，可选的，没有特殊需求可以跳过
        httpProfile = HttpProfile()
        httpProfile.endpoint = "cdn.tencentcloudapi.com"

        # 实例化一个client选项，可选的，没有特殊需求可以跳过
        clientProfile = ClientProfile()
        clientProfile.httpProfile = httpProfile
        # 实例化要请求产品的client对象,clientProfile是可选的
        self.client = cdn_client.CdnClient(cred, "", clientProfile)

    def cdnDomains(self, args) -> None:
        # 查询所有主域名模糊查询
        try:

            # 实例化一个请求对象,每个接口都会对应一个request对象
            req = models.DescribeDomainsConfigRequest()
            params = {
                "Name": "domain",
                "Value": [args['domain']],
                "Fuzzy": True
            }
            req.from_json_string(json.dumps(params))

            # 返回的resp是一个DescribeDomainsConfigResponse的实例，与请求对象对应
            resp = self.client.DescribeDomainsConfig(req)
            # 输出json格式的字符串回包
            obj = resp.to_json_string()
            res = json.loads(obj)
            domain = []
            for d in res['Domains']:
                domain.append(d['Domain'])
            return {"code": 200, "message": domain}
        except TencentCloudSDKException as err:
            print(err.message)
            return {"code": 0, "message": err.message}

    def sslUploadCdn(self, args) -> None:
        # 上传证书到CDN
        try:
            # 实例化一个请求对象,每个接口都会对应一个request对象
            # 新建的CDN需要手动开启
            req = models.UpdateDomainConfigRequest()
            params = {
                "Domain": args['domain'],
                "Https": {
                    "Switch": "on",
                    "Http2": "off",
                    "CertInfo": {
                        "Certificate": args['sslPem'],
                        "PrivateKey": args['sslKey']
                    }
                }
            }
            req.from_json_string(json.dumps(params))

            # 返回的resp是一个UpdateDomainConfigResponse的实例，与请求对象对应
            resp = self.client.UpdateDomainConfig(req)
            # 输出json格式的字符串回包
            print(resp.to_json_string())
            return {"code": 200, "message": 'domain'}
        except TencentCloudSDKException as err:
            print(err.message)
            return {"code": 0, "message": err.message}
